class TypeLablePipeline:
    def __init__(self, data_type):
        self.data_type = data_type

    def apply(self, value):
        result = value.with_columns(
            data_type=self.data_type
        )
        return result